package main

import (
	"fmt"
	"youdu/config"
	"youdu/model"
	"youdu/test"

	"os"
)

var currentDir string

func main() {
	// 获取当前工作目录
	var err error
	currentDir, err = os.Getwd()
	if err != nil {
		fmt.Println("获取当前目录失败:", err)
		return
	}
	// 处理启动参数
	args := os.Args
	argLen := len(args)
	// 配置环境变量
	if argLen > 1 && args[1] == "--dev" {
		config.IsDevMode = true
	}
	// 初始化配置
	err = config.InitConfig()
	if err != nil {
		fmt.Println("init config error:", err)
		return
	}
	err = model.InitDB()
	if err != nil {
		fmt.Println("init db error:", err)
		return
	}
	// 根据参数执行函数
	if argLen > 1 {
		switch args[1] {
		case "--ipa-login":
			test.IpaLogin()
			return
		case "--aes":
			test.Aes()
			return
		case "--ydmsg":
			test.YDMsg()
			return
		}
	}

	startTasks()
	startHttpServer()
}
